package oj;

import java.util.Arrays;

/**
 * Created with IntelliJ IDEA
 * Description
 * User：chenshu
 * Date：2023-10-26
 * Time：11:35
 */
public class TestDemo2 {
    /**
     *
     * @param A
     * @param m 元素个数
     * @param B
     * @param n 元素个数
     */
    public void merge(int A[], int m, int B[], int n) {
        int i = m-1;
        int j = n-1;
        int k = m + n - 1;

        //保证两个数组中都有元素
        while(i >= 0 && j >= 0) {
            if(A[i] > B[j]) {
                A[k] = A[i];
                i--;
                k--;
            } else {
                A[k] = B[j];
                j--;
                k--;
            }
        }

        //补剩下元素
        if(i < 0) {
            while(k >= 0) {
                A[k] = B[j];
                j--;
                k--;
            }
        } else {
            while(k >= 0) {
                A[k] = A[i];
                i--;
                k--;
            }
        }
    }

    public static void main(String[] args) {
        int[] A = {1,0};
        int[] B = {2};
        TestDemo2 test = new TestDemo2();
        test.merge(A,1,B,1);
        System.out.println(Arrays.toString(A));
    }
}
